Pytorch CNN input dimensions not matching

visit shbcf.ru